package algorithm_go

import "fmt"

/*
给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的 两个 整数。

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:
给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
 */
//暴力法
func twoSum1(nums []int, target int) []int {
	for i := 0; i < len(nums); i++ {
		for j := i + 1; j < len(nums); j++ {
			if nums[j] == target-nums[i] {
				return []int{i, j}
			}
		}
	}
	return nil
}

func twoSum2(nums []int, target int) []int {
	result := make([]int, 0)
	tmp := make(map[int]int)
	for i := 0; i < len(nums); i++ {
		tmp[nums[i]] = i
	}

	for i := 0; i < len(nums); i++ {
		complement := target - nums[i]
		if _, ok := tmp[complement]; ok && tmp[complement] != i {
			result = append(result, tmp[complement])
		}
	}
	return result
}

func twoSum3(nums []int, target int) []int {
	tmp := make(map[int]int)
	for i := 0; i < len(nums); i++ {
		complement:= target - nums[i]
		if _,ok := tmp[complement];ok{
			return []int{tmp[complement],i}
		}
		tmp[nums[i]] = i
	}
	return nil
}

func main() {
	//sum := twoSum1([]int{8, 7, 11, 15}, 22)
	//sum := twoSum2([]int{8, 7, 11, 15}, 22)
	sum := twoSum3([]int{8, 7, 11, 15}, 22)
	fmt.Println(sum)
}
